Equim-chan / csu-ems-api

const express    = require('express'),

      superagent = require('superagent'),

      cheerio    = require('cheerio'),

      program    = require('commander'),

      moment     = require('moment'),

      co         = require('co'),

      thunkify   = require('thunkify'),

      access     = require('./lib/access.js');

program

    .option('-h, --help')

    .option('-V, --version')

    .option('-v, --verbose')

    .parse(process.argv);

if (!program.help || !program.version) {

    console.log(

`${`CSUEMS API v${require('./package').version}

by The Liberators`.rainbow}${program.help ? '' :

`

Preparation:

  ${'(This section is WIP)'.grey}

Usage:

  npm start [-- <options...>]

Options:

  -h, --help          print this message and exit.

  -V, --version       print the version and exit.

  -v, --verbose       enable verbose log, by default only errors are logged.

  -p, --port [value]  specify a port to listen, process.env.PORT || 2333 by default.

Examples:

  $ npm start -- -p 43715               # listening to 43715

  $ forever start app.js                # deploy with forever as daemon (root access recommended)

  $ pm2 start -i 0 -n "csuapi" app.js   # deploy with pm2 as daemon  (root access recommended)`}`);

    process.exit(0);

}

superagent.Request.prototype.endThunk = thunkify(superagent.Request.prototype.end);

const port = program.port || process.env.PORT || 2333,

      app = express();

const logging = (log) => console.log(`${moment().format('[[]YY-MM-DD HH:mm:ss[]]')} ${log}`);

const verbose = (log) => program.verbose && logging(log);

// 根据系统时间计算当前学期

const getSem = () => {

    let now = new Date();

    let month = now.getMonth();

    let year = now.getFullYear();

    return month <= 6 ? `${year - 1}-${year}-${month > 0 ? 2 : 1}` : `${year}-${year + 1}-1`;

};

// 获取文档

app.get('/doc', (req, res) => res.sendFile(`${__dirname}/doc/API.html`));

// 查成绩API，通过GET传入用户名和密码

// 查考试API，通过GET传入用户名和密码

app.get(/^\/e(?:|xams)$/, co.wrap(function *(req, res) {

    if (!req.query.id || !req.query.pwd || (req.query.sem && !(/^20\d{2}-20\d{2}-[1-2]$/).test(req.query.sem))) {

        res.status(404).json({ error: "参数不正确" });

        return;

    }

    let start = new Date();

    verbose('Started to query the exams: '.cyan + req.query.id.yellow);

    let headers;

    try {

        headers = yield access.login(req.query.id, req.query.pwd);

    } catch (errMsg) {

        logging(errMsg.inner.red);

        res.status(404).json({ error: errMsg.public });

        return;

    }

    verbose('Successfully logged in.'.green);

    let ires;

    let _sem = req.query.sem || getSem();

    try {

        ires = yield superagent

                         .post('http://csujwc.its.csu.edu.cn/jsxsd/xsks/xsksap_list')

                         .set(headers)

                         .type('form')

                         .send({

                             xqlbmc: '',

                             xnxqid: _sem,

                             xqlb: ''

                         })

                         .endThunk();

    } catch (err) {

        logging(`Failed to reach exams page\n${err.stack}`.red);

        res.status(404).json({ error: '无法进入考试页面' });

        return;

    } finally {

        co(function *() {

            try {

                yield access.logout(headers);

                verbose('Successfully logged out: '.green + req.query.id.yellow);

            } catch (errMsg) {

                logging(errMsg.inner.red);

            }

        });

    }

    verbose('Successfully entered exams page.'.green);

    let $ = cheerio.load(ires.text);

    let top = $('#Top1_divLoginName').text();

    let result = {

        name: top.match(/\s.+\(/)[0].replace(/\s|\(/g, ''),

        id: top.match(/\(.+\)/)[0].replace(/\(|\)/g, ''),

        sem: _sem,

        exams: {},

        'exams-count': 0,

    };

    $('#dataList tr').each(function (index) {

        if (index === 0) {

            return;

        }

        let element = $(this).find('td');

        let title = element.eq(3).text();

        let item = {

            time: element.eq(4).text(),

            location: element.eq(5).text(),

            seat: element.eq(6).text()

        };

        result.exams[title] = item;

        result['exams-count']++;

    });

    res.json(result);

    verbose(`Successfully responded. (req -> res processed in ${new Date() - start}ms)`.green);

}));

app.listen(port, () => {

    logging(`The API is now running on port ${port}. Verbose logging is ${program.verbose ? 'enabled' : 'disabled'}`.green);

});